import importlib
import os
import shutil
import sys
from pdfminer.pdfparser import PDFParser, PDFDocument, PDFEncryptionError
from pdfminer.pdfinterp import PDFResourceManager, PDFPageInterpreter
from pdfminer.converter import PDFPageAggregator
from pdfminer.layout import LTTextBoxHorizontal, LAParams, LTImage
from pdfminer.pdfinterp import PDFTextExtractionNotAllowed
import Zip


importlib.reload(sys)

def fun(pdf_name,i):

    # 源文件
    src_file = Zip.src_dicname + i + '/' + pdf_name
    # 目标文件
    target = Zip.target_dicname + i + '/' + pdf_name.split('.')[0] + '.txt'
    # 错误文件
    wrong = Zip.wrong + i
    # 错误日志
    log = Zip.log + i + '/wrong.txt'


    if not os.path.exists(Zip.target_dicname + i):
        os.makedirs(Zip.target_dicname + i)  # 创建路径

    try:
        # print("处理pdf")
        # 解析PDF文本，并保存到TXT文件中
        fp = open(src_file, 'rb')
        # 用文件对象创建一个PDF文档分析器
        parser = PDFParser(fp)
        # 创建一个PDF文档
        doc = PDFDocument()
        # 连接分析器，与文档对象
        parser.set_document(doc)
        doc.set_parser(parser)

        # 提供初始化密码，如果没有密码，就创建一个空的字符串
        doc.initialize()

        # 检测文档是否提供txt转换，不提供就忽略
        if not doc.is_extractable:
            raise PDFTextExtractionNotAllowed
        else:
            # 创建PDF，资源管理器，来共享资源
            rsrcmgr = PDFResourceManager()
            # 创建一个PDF设备对象
            laparams = LAParams()
            device = PDFPageAggregator(rsrcmgr, laparams=laparams)
            # 创建一个PDF解释器对象
            interpreter = PDFPageInterpreter(rsrcmgr, device)

            # 循环遍历列表，每次处理一个page内容
            # doc.get_pages() 获取page列表
            for page in doc.get_pages():

                interpreter.process_page(page)
                # 接受该页面的LTPage对象
                layout = device.get_result()

                for x in layout:
                    if (isinstance(x, LTTextBoxHorizontal)):  # 获取文本内容
                        with open(target, 'a+', encoding='utf-8', newline='') as f:
                            results = x.get_text()
                            newres = results.replace(" ", "").replace("\t", "").strip()
                            # print(newres)
                            f.write(newres + "\n")

    except BaseException as e:
        if not os.path.exists(Zip.wrong + i):
            os.makedirs(Zip.wrong + i)  # 创建路径

        shutil.move(src_file, wrong + "/")
        with open(log, "a") as file:  # "a"代表追加内容
            file.write(pdf_name + " " + repr(e) + "\n")


